package com.alibaba.system.service.impl;

import com.alibaba.basic.service.impl.BaseServiceImpl;
import com.alibaba.system.domain.Menu;
import com.alibaba.system.mapper.MenuMapper;
import com.alibaba.system.service.IMenuService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 业务实现类：
 */
@Service
public class MenuServiceImpl extends BaseServiceImpl<Menu> implements IMenuService {

    @Autowired
    private MenuMapper menuMapper;

    /**
     * 获取上级部门(部门树)
     * @return
     */
    @Override
    public List<Menu> menuTree() {
        // 1. 一次性查询出所有数据，放到内存中，可以使用map
        // 2. 从所有数据中找出第一层级/顶级部门：这就是我们需要返回的数据
        // 3. 遍历所有部门只要不是第一层级的，找自己的上级，并将自己放入上级的children中
        // 将所有的部门装到map中  - 每次获取父部门都从这里获取
        List<Menu> menus = menuMapper.findAll();
        Map<Long, Menu> menuTemp= new HashMap<>();
        // 将所有部门及其对应的id保存到map中
        for (Menu menu : menus) {
            menuTemp.put(menu.getId(), menu);
        }

        // 前端需要的部门树集合数据
        List<Menu> menuTree = new ArrayList<>();
        // 遍历查询到的所有部门
        for (Menu menu : menus) {
            if (menu.getParentId() == null){ // 如果没有parent_id,则为顶级部门
                menuTree.add(menu);
            } else { // 如果有parent_id
                // 则在map中根据parent_id查询上级部门
                Menu parentMenu = menuTemp.get(menu.getParentId());
                // 将当前部门添加到上级部门的children中
                parentMenu.getChildren().add(menu);
            }
        }
        return menuTree;
    }
}
